package com.dora.basic.juc.threadpool;

import java.util.concurrent.*;

public class ThreadPoolExecutorExample {

    public static void main(String[] args) throws InterruptedException {
        // 配置ThreadPoolExecutor，使用自定义拒绝策略
        ThreadPoolExecutor executor = new ThreadPoolExecutor(
                2, // 核心线程数
                4, // 最大线程数
                60L, // 空闲线程存活时间
                TimeUnit.SECONDS, // 时间单位
                new LinkedBlockingQueue<>(10), // 任务队列（大小为10）
                new CustomRejectedExecutionHandler() // 使用自定义拒绝策略
        );

        // 提交多个任务
        for (int i = 0; i < 20; i++) {
            final int taskId = i;
            executor.submit(() -> {
                try {
                    System.out.println("任务 " + taskId + " 开始执行，线程：" + Thread.currentThread().getName());
                    // 模拟任务执行
                    Thread.sleep(1000);
                    System.out.println("任务 " + taskId + " 执行结束，线程：" + Thread.currentThread().getName());
                } catch (InterruptedException e) {
                    Thread.currentThread().interrupt();
                }
            });
        }

        // 给任务一些时间执行
        Thread.sleep(5000);

        // 关闭线程池
        executor.shutdown();
    }

    // 自定义拒绝策略
    static class CustomRejectedExecutionHandler implements RejectedExecutionHandler {

        @Override
        public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
            // 这里我们选择丢弃任务，并打印日志
            System.out.println("任务被拒绝执行，丢弃任务：" + r.toString());
        }
    }
}
